我們今天會完成CUDA的安裝,以及CUDNN的配置,確認Tensorflow的安裝是否成功,並且可以順利調用GPU。
請到CUDA版本列表尋找所需版本,安裝時請記下安裝位置。
接著下載CUDNN,請確保所下載的CUDNN與CUDA版本匹配,目前請先不要使用CUDNN 8.*.*,先下載7.6.*版本即可,將壓縮檔的所有內容解壓縮至剛剛記下來的安裝位置,預設如下,路徑尾部將隨CUDA版本異動。
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
請輸入Win+Q搜尋環境變數,在其中的系統變數一欄找到Path,編輯Path確保以下所有變數存在,如果你安裝的地方不同,請自行變換變數內容,結束後請重開所有命令提示字元(如果有開的話。
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\libx64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include
請打開Python,確認是否可以正確載入Tensorflow,並且調用到GPU。
>>> import tensorflow
2020-08-31 23:54:36.448690: I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library cudart64_101.dll
你可以安裝jupyter,此編輯器提供的介面與此前提到的Google Colaboratory一致,只是使用的是本機環境。
pip install jupyter
安裝完成後你可以在命令提示字元鍵入以下命令開啟它,你可以透過你的瀏覽器操作,notebook使用時須保持該命令提示字元視窗開啟。
jupyter notebook
如果連結沒有被自動開啟,請手動複製連結到瀏覽器,它會長得像這樣,只是你的token會與我不同,如果沒有複製到token,那就需要你在瀏覽器頁面手動輸入。
http://localhost:8888/?token=04b7447a3cc881995dbc016c24ba63b540d6d5f9dfc277b0
Win10環境下點擊命令提示字元視窗將進入選取模式,進入選取模式將導致jupyter核心暫停,網頁操作沒有回應,請到命令提示字元按ESC離開選取模式,如果要結束核心請按Ctrl+C。